from layers.graph_attention import GraphAttention
from layers.two_layer_mlp import TwoLayerMLP